//
//  Probelm1422.swift
//  TestProject
//
//  Created by 武侠 on 2021/5/18.
//  Copyright © 2021 zhulong. All rights reserved.
//

import UIKit

/*
 1422. 分割字符串的最大得分
 给你一个由若干 0 和 1 组成的字符串 s ，请你计算并返回将该字符串分割成两个 非空 子字符串（即 左 子字符串和 右 子字符串）所能获得的最大得分。
 「分割字符串的得分」为 左 子字符串中 0 的数量加上 右 子字符串中 1 的数量。

 示例 1：
     输入：s = "011101"
     输出：5
     解释：
     将字符串 s 划分为两个非空子字符串的可行方案有：
     左子字符串 = "0" 且 右子字符串 = "11101"，得分 = 1 + 4 = 5
     左子字符串 = "01" 且 右子字符串 = "1101"，得分 = 1 + 3 = 4
     左子字符串 = "011" 且 右子字符串 = "101"，得分 = 1 + 2 = 3
     左子字符串 = "0111" 且 右子字符串 = "01"，得分 = 1 + 1 = 2
     左子字符串 = "01110" 且 右子字符串 = "1"，得分 = 2 + 1 = 3
 示例 2：
     输入：s = "00111"
     输出：5
     解释：当 左子字符串 = "00" 且 右子字符串 = "111" 时，我们得到最大得分 = 2 + 3 = 5
 示例 3：
     输入：s = "1111"
     输出：3
 提示：
     2 <= s.length <= 500
     字符串 s 仅由字符 '0' 和 '1' 组成。
 */
@objcMembers class Probelm1422: NSObject {
    func solution() {
//        print(maxScoreClear("011101"))
//        print(maxScoreClear("00111"))
//        print(maxScoreClear("1111"))
        print(maxScoreClear("00"))
    }
    
    /*
     前缀和
     1: 创建一个数组list[n]，统计字符串中‘0’的前缀和
     2: 遍历list，例如：list[i] (0<i<n-1) 此时如果以i为分割点时，左侧就是list[i] 右侧就是：(n - list[n-1]) - (i - list[i])
     */
    func maxScore(_ s: String) -> Int {
        var list:[Int] = Array(repeating: 0, count: s.count)
        if s.first == "0" {
            list[0] = 1
        }
        let charts:[Character] = Array(s)
        for i in 1..<charts.count {
            if charts[i] == "0" {
                list[i] = list[i-1] + 1
            } else {
                list[i] = list[i-1]
            }
        }
        
        var result = 0
        for i in 0..<list.count-1 {
            result = max(result, list[i] + (s.count - list.last!) - (i + 1 - list[i]))
        }
        return result
    }
    
    /*
     优化空间
     1: 遍历一次字符串，统计1的总个数 sum1, 那么0的总个数就是 n - sum1
     2: 遍历字符串，例如在i的位置：0的个数时left0，那么1的个数就是i+1-left0，那么右侧1的个数: sum1 - (i+1-left0)
     */
    func maxScoreClear(_ s: String) -> Int  {
        // 1: 统计1的个数
        var sum1 = 0
        for c in s {
            if c == "1" {
                sum1 += 1
            }
        }
        var result = 0
        // 统计左侧0的个数
        var left0 = 0
        for (i, c) in s.enumerated() {
            if i == s.count - 1 {
                break
            }
            if c == "0" {
                left0 += 1
            }
            
            print(left0, i+1, sum1)
            // 右侧1的个数：sum1 - (i+1-left0)
            result = max(result, left0 + sum1 - (i+1-left0))
        }
        return result
    }
}
